home *** CD-ROM | disk | FTP | other *** search
/ Visual Basic Source Code / Visual Basic Source Code.iso / vbsource / vbdatabs / twalk.cpp < prev    next >
C/C++ Source or Header  |  1999-03-14  |  4KB  |  134 lines
  1. // ------------------------------- //
  2. // -------- Start of File -------- //
  3. // ------------------------------- //
  4. // ----------------------------------------------------------- // 
  5. // C++ Source Code File Name: twalk.cpp
  6. // Compiler Used: MSVC40, DJGPP 2.7.2.1, GCC 2.7.2.1, HP CPP 10.24
  7. // Produced By: Doug Gaer 
  8. // File Creation Date: 01/23/1997 
  9. // Date Last Modified: 03/15/1999
  10. // Copyright (c) 1997 Douglas M. Gaer
  11. // ----------------------------------------------------------- // 
  12. // ------------- Program Description and Details ------------- // 
  13. // ----------------------------------------------------------- // 
  14. /*
  15. The VBD C++ classes are copyright (c) 1997, by Douglas M. Gaer.
  16. All those who put this code or its derivatives in a commercial
  17. product MUST mention this copyright in their documentation for
  18. users of the products in which this code or its derivative
  19. classes are used. Otherwise, you have the freedom to redistribute
  20. verbatim copies of this source code, adapt it to your specific
  21. needs, or improve the code and release your improvements to the
  22. public provided that the modified files carry prominent notices
  23. stating that you changed the files and the date of any change.
  24.  
  25. THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND.
  26. THE ENTIRE RISK OF THE QUALITY AND PERFORMANCE OF THIS SOFTWARE
  27. IS WITH YOU. SHOULD ANY ELEMENT OF THIS SOFTWARE PROVE DEFECTIVE,
  28. YOU WILL ASSUME THE COST OF ALL NECESSARY SERVICING, REPAIR, OR
  29. CORRECTION.
  30.  
  31. The TreeWalkerb class defines a generic iterator used to walk
  32. through (R)ed (B)lack binary search trees. All of the member
  33. functions are accessed via pointers to the member functions.
  34. */
  35. // ----------------------------------------------------------- // 
  36. #include "twalk.h"
  37.  
  38. void TreeWalkerb::Reset(BNodeBase *r, WalkOrder w)
  39. {
  40.   root = r;
  41.   worder = w;
  42.   if (worder == PREORDER) NextFptr = &TreeWalkerb::NextPreOrder;
  43.   else if (worder == INORDER) NextFptr = &TreeWalkerb::NextInOrder;
  44.   else if (worder == POSTORDER) NextFptr = &TreeWalkerb::NextPostOrder;
  45.   else NextFptr = &TreeWalkerb::NextLvlOrder;
  46.   state = 0;
  47.   curr = root;
  48.   path.Clear();
  49.   if (root && worder == LVLORDER) path.Insert(root);
  50. }
  51.  
  52. BNodeBase *TreeWalkerb::NextPreOrder()
  53. {
  54.   while(1) {
  55.     if (curr) {
  56.        path.Push(curr);
  57.        BNodeBase *p = curr;
  58.        curr = curr->Left;
  59.        return p;
  60.     }
  61.     else {
  62.       if (path.Pop(curr) == 0) {
  63.          return curr = 0;
  64.       }
  65.       else curr = curr->Right;
  66.     }
  67.   }
  68. }
  69.  
  70. BNodeBase *TreeWalkerb::NextInOrder()
  71. {
  72.   while(1) {
  73.     if (curr) {
  74.        path.Push(curr);
  75.        curr = curr->Left;
  76.     }
  77.     else {
  78.       if (path.Pop(curr) == 0) {
  79.          return curr = 0;
  80.       }
  81.       else {
  82.          BNodeBase *p = curr;
  83.          curr = curr->Right;
  84.          return p;
  85.       }
  86.     }
  87.   }
  88. }
  89.  
  90. BNodeBase *TreeWalkerb::NextPostOrder()
  91. {
  92.   while(1) {
  93.     if (state == 0) { // Ready to go down the tree to left
  94.        if (curr) {
  95.          path.Push(curr);
  96.           curr = curr->Left;
  97.        }
  98.        else state = 1;
  99.     }
  100.     else { // State == 1: // Ready to come up the tree
  101.        BNodeBase *c = curr;
  102.        if (path.IsEmpty()) return curr = 0; // At root
  103.        curr = *path.Top();
  104.        if (c == curr->Left && curr->Right) {
  105.           // Coming back up the tree from the left, and
  106.           // there is a right child, so go right. 
  107.           // Note that curr is still on top of stack.
  108.           curr = curr->Right;
  109.           state = 0;
  110.        }
  111.        else {
  112.           // Coming back up the tree from the right,
  113.           // or there was no right child, so visit
  114.           // the node, and continue on up. (State
  115.           // stays at 1.)
  116.           path.Pop();
  117.           return curr;
  118.        }
  119.     }
  120.   }
  121. }
  122.  
  123. BNodeBase *TreeWalkerb::NextLvlOrder()
  124. {
  125.   if (path.Extract(curr) == 0) return 0;
  126.   if (curr->Left != 0)  path.Insert(curr->Left);
  127.   if (curr->Right != 0) path.Insert(curr->Right);
  128.   return curr;
  129. }
  130. // ----------------------------------------------------------- // 
  131. // ------------------------------- //
  132. // --------- End of File --------- //
  133. // ------------------------------- //
  134.